package com.cy.ragbase.dto;

import lombok.Data;
import java.util.List;

@Data
public class CodeRagResponse {
    /**
     * 原始需求
     */
    private String originalRequirement;
    
    /**
     * 生成的代码提示词
     */
    private String generatedPrompt;
    
    /**
     * 参考代码块
     */
    private List<ReferenceCode> referenceCodes;
    
    /**
     * 总检索数量
     */
    private int totalRetrieved;
    
    /**
     * 处理时间
     */
    private long processingTimeMs;
    
    @Data
    public static class ReferenceCode {
        /**
         * 代码块ID
         */
        private String id;
        
        /**
         * 模块名称
         */
        private String name;
        
        /**
         * 代码内容
         */
        private String code;
        
        /**
         * 描述信息
         */
        private String info;
        
        /**
         * 标签
         */
        private List<String> labels;
        
        /**
         * 编程语言
         */
        private String language;
        
        /**
         * 相似度分数
         */
        private float score;
    }
} 